/*
 * @lc app=leetcode.cn id=450 lang=typescript
 *
 * [450] 删除二叉搜索树中的节点
 */

// @lc code=start
/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */

//  思路：迭代
//  要被删除的节点主要分为3种情况：
//  1.无子节点，删除即可
//  2.有单个子节点，子节点接替即可
//  3.有两个子节点，需要找到左侧最大或右侧最小的节点接替
//  参考：https://labuladong.github.io/algo/2/19/26/

//  复杂度：O(logn) O(n)

function deleteNode(root: TreeNode | null, key: number): TreeNode | null {
    if (!root) return null
    if (root.val === key) {
        //  处理情况1，2
        if (!root.right) return root.left
        if (!root.left) return root.right
        //  处理情况3
        const minNode = getMin(root.right)
        root.val = minNode.val
        root.right = deleteNode(root.right, minNode.val)
    } else if (key < root.val) {
        root.left = deleteNode(root.left, key)
    } else if (key > root.val) {
        root.right = deleteNode(root.right, key)
    }
    return root
};

function getMin(node: TreeNode): TreeNode {
    while (node.left) node = node.left
    return node
}
// @lc code=end

import { TreeNode } from "./type"
const root = TreeNode.buildTree([5, 3, 6, 2, 4, null, 7])
console.log(TreeNode.printTree(deleteNode(root, 3)))